# 导入相关的模块
# matplotlib模块用于图像的显示
# torchvision库用于进行数据增强
# PIL库用于图像的读取
import matplotlib.pyplot as plt
import torchvision.transforms as T
from PIL import Image

# 指定一下四张图片的路径位置
path = ["../pic/sign1.jpg", "../pic/sign2.jpg", "../pic/sign3.jpeg", "../pic/sign4.jpeg"]


# 循环读取每一张图片，得到index为索引号，p为图片的文件路径信息
for index, p in enumerate(path):
    plt.subplot(2, 2, 1+index) # 绘制第2行第2列的第index+1个图片
    img_array = Image.open(p) # 得到第index+1个图片的数组
    # 利用torchvision.transforms.ColorJitter(brightness=, contrast=, saturation=, hue=)方法进行图像亮度、对比度、颜色随机变化
    new_img_array = T.ColorJitter(brightness=1)(img_array) # 传入参数1，亮度在0-2之间变化
    # new_img_array = T.ColorJitter(contrast=1)(img_array)  # 传入参数1，对比度在0-2之间变化
    plt.imshow(new_img_array) # 画出图像
plt.show() # 显示整体四张图片
